tensor([[[0.3500, 0.3500],
         [0.3500, 0.3500]],

        [[0.3000, 0.3000],
         [0.3000, 0.3000]],

        [[0.3400, 0.3400],
         [0.3400, 0.3400]]], grad_fn=<RoundBackward1>)